from api.genggai2fenjing_operation import genggai2fenjing_intro, fenjing_intro2fenjing_prompt, \
    fenjing_prompt2fenjing_localurl, fenjing2fenjing_prompt, fenjing2fenjing_localurl


#input：故事的中文梗概
#ōutput：一个包含n个元素的列表，每个元素是一个元组，元组的第一个元素是分镜中文介绍，第二个元素是分镜prompt，第三个元素是分镜图片的相对饮用位置，第四个元素是分镜的任务id
#exp：[[分镜1中文介绍,分镜1prompt,分镜1的相对引用位置,分镜1的任务id],[...],[...]]

def genggai2fenjing(genggai):
    list_fenjing_intro = genggai2fenjing_intro(genggai)
    print("分镜中文介绍：", list_fenjing_intro)
    list_fenjing_prompt = fenjing_intro2fenjing_prompt(list_fenjing_intro)
    print("分镜prompt：", list_fenjing_prompt)
    # list_fenjing_localurl = fenjing_prompt2fenjing_localurl(list_fenjing_prompt)
    # print("分镜localurl：", list_fenjing_localurl)
    # 将三个的列表合并成一个列表，每个list的第n个元素合并成一个元组
    list_allinone = []
    for i in range(len(list_fenjing_intro)):
        list_allinone.append(
            [list_fenjing_intro[i], list_fenjing_prompt[i],
             # list_fenjing_localurl[i][0], list_fenjing_localurl[i][2]
             ])
    return list_allinone


def fenjing2localurl(genggai):
    # list_fenjing_intro = genggai2fenjing_intro(genggai)
    # print("分镜中文介绍：", list_fenjing_intro)
    list_fenjing_prompt = fenjing2fenjing_prompt(genggai)
    print("分镜prompt：", list_fenjing_prompt)
    list_fenjing_localurl = fenjing2fenjing_localurl(list_fenjing_prompt)
    # print("分镜localurl：", list_fenjing_localurl)
    # 将三个的列表合并成一个列表，每个list的第n个元素合并成一个元组
    return list_fenjing_localurl[1], list_fenjing_localurl[2]